package servlet;

import org.apache.commons.io.IOUtils;
import sun.misc.BASE64Encoder;

import javax.servlet.ServletContext;
import javax.servlet.ServletException;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.InputStream;
import java.net.URLEncoder;

//文件下载
public class DownloadServlet extends HttpServlet {

    protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        //1.获取下载的文件名
        String downloadFileName = "b.jpg";

        //2.读取要下载的文件内容（通过servletContext对象可以读取）
        ServletContext servletContext = getServletContext();

        //3.获取要下载的文件类型
        String mimeType = servletContext.getMimeType("/file/" + downloadFileName);
        System.out.println("下载的文件类型: " + mimeType);

        //4.在回传前，通过响应头告诉客户端返回的数据类型
        response.setContentType(mimeType);

        //5.还要告诉客户端收到的数据时用于下载使用（还是响应头）
        // Content-Disposition 响应头，表示收到的数据怎么处理
        // attachment 表示附件，也就是下载的一个文件。
        // fileName=后面，表示下载的文件名。
        // URL编码是把汉字转换成%xx%xx的格式(解决下载文件中文名乱码问题)
        // 使用请求头中User-agent来判断浏览器类型
        String ua = request.getHeader("User-Agent");
        if (ua.contains("Firefox")) {
            //使用下面格式进行BASE64编码后
            String str2 = "attachment; fileName=" + "=?utf-8?B?"
                    + new BASE64Encoder().encode("中文.jpg".getBytes("utf-8")) + "?=";
            //设置到响应头中
            response.setHeader("Content-Disposition", str2);
        } else {
            //把中文名进行UTF-8编码操作
            String str = "attachment; filename = " + URLEncoder.encode("中国.jpg", "UTF-8");
            //然后把编码后的字符串设置到响应头中
            response.setHeader("Content-Disposition", str);
        }

        //6.根据数据流获取文件
        /* /斜杠被服务器解析表示地址为http://id:port/工程名 */
        InputStream resourceAsStream = servletContext.getResourceAsStream("/file/" + downloadFileName);
        //获取响应的输出流
        ServletOutputStream outputStream = response.getOutputStream();
        //读取输入流中的全部数据，复制给输出流，输出给客户端（使用commons-io包来操作数据流 ）
        IOUtils.copy(resourceAsStream, outputStream);
    }
}
